package com.ruifu.doc.utils;

import com.ruifu.common.utils.AssignorsUtils;
import com.ruifu.doc.conf.FileConfig;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import java.util.Map;
import java.util.Set;
@Component
public class GetPreviewFileTypeUtils {
    /**
     * 文件操作配置类
     */
    @Autowired
    private FileConfig fileConfig;
    /**
     * 根据文件类型获取要转换的预览文件类型
     *
     * @param sourceType
     * @param conversionMap
     * @return
     */
    public String getPreviewFileType(String sourceType, Map<String, String> conversionMap) {
        //获取转换map中的所有转换后的类型
        Set<String> fileTypeSet = conversionMap.keySet();
        //声明转换后的文件类型
        String previewFileType = null;
        for (String fileType : fileTypeSet) {
            if (AssignorsUtils.isContains(fileConfig.getConversionMap().get(fileType), sourceType)) {
                previewFileType = fileType;
                break;
            }
        }
        return previewFileType;
    }
}
